package com.zzyl.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.date.LocalDateTimeUtil;
import cn.hutool.core.text.CharSequenceUtil;
import cn.hutool.core.util.NumberUtil;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.zzyl.base.PageResponse;
import com.zzyl.constant.CacheConstants;
import com.zzyl.dto.AlertRuleDto;
import com.zzyl.entity.AlertData;
import com.zzyl.entity.AlertRule;
import com.zzyl.mapper.AlertDataMapper;
import com.zzyl.mapper.AlertRuleMapper;
import com.zzyl.mapper.DeviceMapper;
import com.zzyl.mapper.UserRoleMapper;
import com.zzyl.service.AlertRuleService;
import com.zzyl.utils.ObjectUtil;
import com.zzyl.utils.StringUtils;
import com.zzyl.vo.AlertRuleVo;
import com.zzyl.vo.DeviceDataVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.temporal.ChronoUnit;
import java.util.Collection;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * @Description AlertRuleServiceImpl
 * @Author mks
 * @Date 2024-09-14
 */
@Slf4j
@Service
public class AlertRuleServiceImpl implements AlertRuleService {

    @Autowired
    private AlertRuleMapper alertRuleMapper;

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Autowired
    private DeviceMapper deviceMapper;

    @Autowired
    private UserRoleMapper userRoleMapper;

    @Autowired
    private AlertDataMapper alertDataMapper;

    //设备维护人员的角色名称
    @Value("${zzyl.alert.deviceMaintainerRole}")
    private String deviceMaintainerRole;
    //超级管理员的角色名称
    @Value("${zzyl.alert.managerRole}")
    private String managerRole;

    /**
     * 新增报警规则
     * @param alertRuleDto
     */
    @Override
    public void addAlertRule(AlertRuleDto alertRuleDto) {
        AlertRule alertRule = BeanUtil.toBean(alertRuleDto, AlertRule.class);
        alertRuleMapper.insert(alertRule);
    }

    /**
     * 分页获取警告规则列表
     * @param pageNum
     * @param pageSize
     * @param alertRuleName
     * @param functionName
     * @param productKey
     * @return
     */
    @Override
    public PageResponse<AlertRuleVo> getPage(Integer pageNum, Integer pageSize, String alertRuleName, String functionName, String productKey) {

        PageHelper.startPage(pageNum,pageSize);
        Page<AlertRuleVo> page = alertRuleMapper.getPage(alertRuleName, functionName, productKey);
        // 补充报警规则 rules
        page.getResult().forEach(alertRuleVo -> {
            String alertReason = CharSequenceUtil.format("{}{}{},持续{}个周期就报警",
                    alertRuleVo.getFunctionName(),
                    alertRuleVo.getOperator(),
                    alertRuleVo.getValue(),
                    alertRuleVo.getDuration());
            alertRuleVo.setRules(alertReason);
        });
        return PageResponse.of(page,AlertRuleVo.class);
    }

    /**
     * 获取单个告警规则
     * @param id
     * @return
     */
    @Override
    public AlertRuleVo getAlertRuleId(Long id) {
        AlertRule alertRule = alertRuleMapper.selectByPrimaryKey(id);

        return BeanUtil.toBean(alertRule,AlertRuleVo.class);
    }

    /**
     * 更新告警规则
     * @param id
     * @param alertRuleDto
     */
    @Override
    public void updateAlertRuleId(Long id,AlertRuleDto alertRuleDto) {
        AlertRule alertRule = BeanUtil.toBean(alertRuleDto, AlertRule.class);
        alertRuleMapper.updateByPrimaryKeySelective(alertRule);
    }

    /**
     * 启用-禁用告警
     * @param id
     * @param status
     */
    @Override
    public void updateAlertRuleStatus(Long id, Integer status) {
        alertRuleMapper.updateAlertRuleStatus(id,status);
    }

    /**
     * 删除告警规则
     * @param id
     */
    @Override
    public void deleteAlertRuleId(Long id) {
        alertRuleMapper.deleteAlertRuleId(id);
    }


    /*-----------------    报警数据过滤及处理    -----------------*/
    /**
     * 校验设备上报数据，进行报警规则过滤处理
     *
     * @param deviceDataVo 设备上报数据
     */
    @Override
    public void alertFilter(DeviceDataVo deviceDataVo) {
        // 1.获取设备上报数据时间，如果上报发生在1分钟前，为历史数据，不再处理
        LocalDateTime alarmTime = deviceDataVo.getAlarmTime();
        // 1.1 计算设备上报数据时间与当前时间的间隔
        // ChronoUnit.SECONDS表示计算两个时间之间的间隔单位是秒。
        long between = LocalDateTimeUtil.between(alarmTime, LocalDateTime.now(), ChronoUnit.SECONDS);
        // 1.2 如果上报发生在1分钟前，为历史数据，程序结束
        if (between > 60) {
            return;
        }

        // 2.查询报警规则，将对于全部设备的报警规则 和 对于本设备的报警规则合并
        // 2.1 查询全部设备的报警规则
        List<AlertRule> allDeviceAlertRules = alertRuleMapper.selectByFunctionId(deviceDataVo.getFunctionId(), "-1", deviceDataVo.getProductKey());
        // 2.2 查询本设备的报警规则
        List<AlertRule> theDeviceAlertRules = alertRuleMapper.selectByFunctionId(deviceDataVo.getFunctionId(), deviceDataVo.getIotId(), deviceDataVo.getProductKey());
        Collection<AlertRule> alertRules = CollUtil.addAll(allDeviceAlertRules, theDeviceAlertRules);

        // 3.3 如果报警规则为空，程序结束
        if (CollectionUtil.isEmpty(alertRules)) {
            return;
        }

        // 4. 如果报警规则不为空，遍历报警规则，进行校验
        alertRules.forEach(alertRule -> deviceDateAlarmHandler(deviceDataVo, alertRule));
    }

    /**
     * 设备数据报警处理器
     *
     * @param deviceDataVo 设备上报数据
     * @param alertRule    报警规则
     */
    private void deviceDateAlarmHandler(DeviceDataVo deviceDataVo, AlertRule alertRule) {
        // 1.第一层校验：判断是否在生效时段内
        String[] aepArr = alertRule.getAlertEffectivePeriod().split("~");// 1.1 获取报警生效的时段
        LocalTime startTime = LocalTime.parse(aepArr[0]); // 1.1.1 获取报警生效的时段的起始时间
        LocalTime endTime = LocalTime.parse(aepArr[1]); // 1.1.1 获取报警生效的时段的终止时间
        // 1.2 获取上报时间
        LocalTime time = LocalDateTimeUtil.of(deviceDataVo.getAlarmTime()).toLocalTime();

        // 1.3 判断上报时间是否在生效时间内，如果上报时间不在生效时间段内，校验结束
        if (startTime.isAfter(time) || endTime.isBefore(time)) {
            return;
        }

        // 获取设备id
        String iotId = deviceDataVo.getIotId();
        // 从redis中获取数据，设置报警规则连续触发次数的键名(要唯一值)rediskey
        String aggCountKey = CacheConstants.ALERT_TRIGGER_COUNT_PREFIX +
                iotId + ":" + deviceDataVo.getFunctionId() + ":" + alertRule.getId();

        // 2. 第二层校验：将设备上报值和阈值比较
        // x – 第一个值，y – 第二个值。x==y返回0，x<y返回小于0的数，x>y返回大于0的数
        int compareResult = NumberUtil.compare(Double.parseDouble(deviceDataVo.getDataValue()), alertRule.getValue());
        //或 逻辑运算，2个条件满足其一即为true
        //1.运算符为>=，且设备上报值 >= 阈值
        //2.运算符为<，且设备上报值 < 阈值
        if ((ObjectUtil.equals(alertRule.getOperator(), ">=") && compareResult >= 0)
                || (ObjectUtil.equals(alertRule.getOperator(), "<") && compareResult < 0)) {
            log.debug("此时设备上报数据达到阈值");
        } else {
            //该情况不符合报警规则，所以设备上报数据为正常数据。需要删除redis聚合的异常数据,程序结束
            redisTemplate.delete(aggCountKey);
            return;
        }

        // 3.第三层校验：校验当前设备+功能的报警规则是否处于沉默周期，设置沉默周期键名(要唯一值)rediskey
        String silentCacheKey = CacheConstants.ALERT_SILENT_PREFIX + iotId + ":" + deviceDataVo.getFunctionId() + ":" + alertRule.getId();
        String silentData = redisTemplate.opsForValue().get(silentCacheKey);// 获取缓存中的沉默周期数据
        //  3.1 如果redis中存在对应沉默周期数据，则当前设备+功能正处于沉默周期，无需报警
        if (StringUtils.isNotEmpty(silentData)) {
            return;
        }

        // 4.第四层校验：校验持续周期

        // 4.1获取连续触发报警规则的次数
        // aggCountData是上次触发报警累计到的次数，这一次又触发了报警，所以累积到这次就+1
        String aggCountData = redisTemplate.opsForValue().get(aggCountKey);// 获取在本次触发报警之前累计到的次数
        int triggerCount = ObjectUtil.isEmpty(aggCountData) ? 1 : Integer.parseInt(aggCountData) + 1;

        // 4.2判断次数是否与持续周期相等
        //如果触发报警规则次数不等于持续周期，则无需报警
        if (ObjectUtil.notEqual(alertRule.getDuration(), triggerCount)) {
            redisTemplate.opsForValue().set(aggCountKey, triggerCount + "");
            return;
        }

        // 5.如果触发报警规则次数等于持续周期，则需要报警，完成以下操作：
        redisTemplate.delete(aggCountKey);//1）删除报警规则连续触发次数的缓存
        //2）添加对应的沉默周期，设置过期时间，添加对应的沉默周期，设置过期时间
        redisTemplate.opsForValue().set(silentCacheKey, deviceDataVo.getDataValue(), alertRule.getAlertSilentPeriod(), TimeUnit.MINUTES);


        // 6.查询护理员(设备维护人员)
        List<Long> consumerIds = null;
        if (ObjectUtil.equals(0, alertRule.getAlertDataType())) {
            //如果是老人报警数据，需要通知绑定老人的护理员。根据iotId查询老人绑定的护理员
            if(deviceDataVo.getLocationType() == 0){
                consumerIds = deviceMapper.selectNursingIdsByIotIdWithElder(iotId);
            }else if(deviceDataVo.getLocationType() == 1 && deviceDataVo.getPhysicalLocationType() == 2){
                consumerIds = deviceMapper.selectNursingIdsByIotIdWithBed(iotId);
            }
        } else {
            //如果是设备报警数据，需要通知设备维护人员。根据指定角色名称查询相关用户
            consumerIds = userRoleMapper.selectUserIdsByRoleName(deviceMaintainerRole);
        }
        //查询超级管理员，超级管理员无论什么消息都会接收
        List<Long> managerIds = userRoleMapper.selectUserIdsByRoleName(managerRole);
        List<Long> allConsumerIds = CollUtil.addAllIfNotContains(consumerIds, managerIds);
        allConsumerIds = CollUtil.distinct(allConsumerIds);

        // 7.报警数据存储到数据库，新增报警数据
        insertAlertData(deviceDataVo, alertRule, allConsumerIds);
    }

    /**
     * 新增报警数据
     * @param deviceDataVo 设备上报数据
     * @param alertRule    报警规则
     * @param consumerIds  消费者id列表
     * @return 报警数据
     */
    private void insertAlertData(DeviceDataVo deviceDataVo, AlertRule alertRule, List<Long> consumerIds) {
        // 1.1 生成一个描述报警原因的字符串
        String alertReason = CharSequenceUtil.format(
                "{}{}{},持续{}个周期就报警",
                alertRule.getFunctionName(),
                alertRule.getOperator(),
                alertRule.getValue(),
                alertRule.getDuration());
        // 1.2将DeviceDataVo转化成AlertData
        AlertData alertData = BeanUtil.toBean(deviceDataVo, AlertData.class);
        // 1.3补充基础信息
        alertData.setAlertRuleId(alertRule.getId());
        alertData.setAlertReason(alertReason);
        alertData.setType(alertRule.getAlertDataType());
        alertData.setStatus(0);

        // 封装通知报警数据集合
        List<AlertData> list = consumerIds.stream().map(id -> {
            AlertData dbAlertData = BeanUtil.toBean(alertData, AlertData.class);
            dbAlertData.setUserId(id);// 设置当前的用户(如护理员/管理员/维修工)
            return dbAlertData;
        }).collect(Collectors.toList());
        // 批量插入
        alertDataMapper.batchInsert(list);
    }
}
